R/turnbull add speaker.R

In trendlock/speeches: Tidytext analysis of Malcom Turnbull's and Bill Shorten's speeches

#' Add speaker categorical variable
#'
#' Mutates variable \code{"speaker."} to either \code{"jorn"} or \code{"pm"}.
#' Also removes the sentences that are \code{"JOURNALIST:"} and \code{"PRIME MINISTER:"}.
#'
#' @name clean_text
#' @param df Takes df with sentence level text.
#' @param keep Defaults to TRUE, if FALSE filters out "jorn"...
#' @return Returns data frame
#' @export


add_speaker <- function(df, keep = T) {
  df <- df %>%
    mutate(
      sent = str_trim(sent),
      speaker. = case_when(sent == "JOURNALIST:" ~ "Journalist",
                           sent == "PRIME MINISTER:" ~ "Malcolm Turnbull"))

  df <- df %>%
    fill(speaker.) %>% # annying but this line not working unless split from the call above??
    mutate(speaker. = case_when(is.na(speaker.) ~ "pm",
                                !is.na(speaker.) ~ speaker.)) %>%
    filter(!sent %in% c("JOURNALIST:", "PRIME MINISTER:"))


  if (keep){
    df
  } else {
    df %>%
      filter(speaker. == "pm")
  }
}